package com.domain.dto;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.Data;

import java.util.Collections;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * 封装分页结果
 * @author 17645
 * @date 2024/06/17
 */
@Data
public class PageDTO<T> {
    /**
     * 总条数
     */
    private Long total;
    /**
     * 总页数
     */
    private Long pages;
    /**
     * 查询结果
     */
    private List<T> list;

    public static <VO,PO> PageDTO<VO> of(Page<PO> page,Class<VO> clazz) {
        //封装Vo
        PageDTO<VO> pageDTO = new PageDTO<>();
        pageDTO.setTotal(page.getTotal());
        pageDTO.setPages(page.getPages());
        List<PO> records = page.getRecords();
        if (CollUtil.isEmpty(records)) {
            pageDTO.setList(Collections.emptyList());
        }
        List<VO> userVos = BeanUtil.copyToList(records, clazz);
        pageDTO.setList(userVos);
        return pageDTO;
    }

    public static <VO,PO> PageDTO<VO> of(Page<PO> page, Function<PO,VO> convertor) {
        //封装Vo
        PageDTO<VO> pageDTO = new PageDTO<>();
        pageDTO.setTotal(page.getTotal());
        pageDTO.setPages(page.getPages());
        List<PO> records = page.getRecords();
        if (CollUtil.isEmpty(records)) {
            pageDTO.setList(Collections.emptyList());
        }
        pageDTO.setList(records.stream().map(convertor).collect(Collectors.toList()));
        return pageDTO;
    }
}
